"""
垃圾邮件检测器配置文件
"""

# 模型配置 
MODEL_CONFIG = {
    'max_features': 5000,
    'ngram_range': (1, 2),
    'min_df': 1,
    'max_df': 0.95
}

# 文件路径 
PATHS = {
    'data': 'data/spam_data.csv',
    'model': 'model/spam_model.pkl',
    'vectorizer': 'model/vectorizer.pkl'
}

# 邮件类别 
LABELS = {
    'ham': 0,
    'spam': 1
}

# 停用词列表（可以自定义）
CHINESE_STOP_WORDS = [
    '的', '了', '在', '是', '我', '有', '和', '就', '不', '人', '都', '一', '一个', '上', '也', '很', '到', '说', '要', '去', '你',
    '会', '着', '没有', '看', '好', '自己', '这', '那', '来', '用', '他', '她', '它', '们', '个', '啊', '吧', '呢', '吗', '嗯',
    '哦', '哈', '嘿', '呀', '哟', '呵', '唉', '嗯嗯', '哼', '呜', '哇', '咦', '咯', '咚', '咔', '咳', '咪', '咭', '咻', '咿',
    '为', '而', '如', '因', '所以', '但是', '然而', '可是', '虽然', '尽管', '无论', '不管', '无', '没', '还', '又', '才', '便',
    '这样', '那样', '这里', '那里', '这些', '那些', '什么', '怎么', '为什么', '怎样', '多少', '几', '第', '每', '各', '另',
    '其', '其他', '别', '别的', '谁', '哪', '哪里', '哪些', '何', '何时', '何地', '何人', '如何', '多么', '何等', '多大'
]

# 示例垃圾邮件和正常邮件（用于测试和扩展数据集）
spam_emails = [
    "恭喜您！您已中奖100万元！立即拨打电话领取大奖！千万不要错过！",
    "免费获得减肥药！一周瘦20斤！效果保证！不瘦退款！立即订购！",
    "紧急通知！您的银行账户即将被冻结！请立即点击链接验证身份！",
    "投资理财！月收益30%！无风险！保本保息！立即加入！",
    "免费送苹果手机！限时优惠！点击领取！机会难得！",
    "代开发票！增值税普票专票！价格优惠！质量保证！",
    "网络兼职！在家赚钱！日入300元！无需经验！",
    "股票内幕消息！明日涨停板！千载难逢机会！立即购买！",
    "办理贷款！无抵押！当天放款！利息低！额度高！",
    "减肥神器！一个月瘦30斤！明星同款！效果惊人！",
    "免费试用化妆品！美白祛斑！立竿见影！包邮到家！",
    "彩票预测！百分百中奖！专业团队！成功率极高！",
    "赚钱机会！躺着就能赚钱！月入过万！立即加入！",
    "免费领取红包！无门槛！立即到账！点击领取！",
    "澳门赌场！网上百家乐！必赢秘籍！立即下载！",
    "办证刻章！身份证护照！质量保证！价格合理！",
    "股票推荐！内幕消息！稳赚不赔！立即购买！",
    "免费抽奖！豪车别墅！点击参与！中奖率极高！",
    "医疗广告！根治绝症！包治包好！立即咨询！",
    "投资项目！高回报！低风险！月收益50%！"
]
        
normal_emails = [
    "明天下午3点开会，请大家准时参加。会议室在A栋302。",
    "您好，您的快递已经到达，请及时取件。快递单号：SF123456789。",
    "生日快乐！祝您身体健康，工作顺利，家庭幸福！",
    "会议纪要已发送，请查收附件并于明日前反馈意见。",
    "感谢您参加今天的培训，课件资料已上传到公司网盘。",
    "您的机票预订成功，航班号CA1234，明天早上8点起飞。",
    "项目进度汇报：目前已完成60%，预计下周完成全部工作。",
    "年终总结报告已提交，请领导审核批准。",
    "客户反馈很好，对我们的服务非常满意。",
    "明天的聚餐改到晚上7点，地点还是老地方。",
    "报销单据已整理完毕，请财务部门审核。",
    "产品发布会定于下周五举行，请提前准备相关材料。",
    "学术会议通知：计算机学会年会将于下月举行。",
    "图书馆新书到馆通知，欢迎大家前来借阅。",
    "公司年会节目征集，请大家积极报名参与。",
    "健康体检通知：请于本月内完成年度体检。",
    "工资单已发放，请查看邮箱附件。",
    "新员工入职培训安排在下周二上午。",
    "系统维护通知：明晚8点至12点系统将暂停服务。",
    "感谢您的订阅，我们会定期发送最新资讯。"
]